# # base
# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain amazon --target-domain webcam \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain dslr --target-domain webcam \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain webcam --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain webcam --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# tiny office 31
# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain amazon --target-domain webcam \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain dslr --target-domain webcam \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain webcam --target-domain amazon \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_vmamba \
#     --dataset Office31 --source-domain webcam --target-domain dslr \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# # base office home
# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome \
#     --dataset OfficeHome --source-domain Art --target-domain Clipart \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Art --target-domain Product \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Art --target-domain RealWorld \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Clipart --target-domain Art \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Clipart --target-domain Product \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Clipart --target-domain RealWorld \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Product --target-domain Art \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Product --target-domain Clipart \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain Product --target-domain RealWorld \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain RealWorld --target-domain Art \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain RealWorld --target-domain Clipart \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome  \
#     --dataset OfficeHome --source-domain RealWorld --target-domain Product \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# # base office home
# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm VMamba --output ./output/train_output_with_vmamba/officehome \
#     --dataset OfficeHome --source-domain Art --target-domain Clipart \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

#tiny office home
CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1 \
    --dataset OfficeHome --source-domain Art --target-domain Clipart \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Art --target-domain Product \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Art --target-domain RealWorld \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Clipart --target-domain Art \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Clipart --target-domain Product \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Clipart --target-domain RealWorld \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Product --target-domain Art \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Product --target-domain Clipart \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain Product --target-domain RealWorld \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain RealWorld --target-domain Art \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain RealWorld --target-domain Clipart \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11776 \
    main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    --data-path ../dataset/OfficeHome \
    --algorithm VMamba --output ./output/train_output_with_vmamba/officehome1  \
    --dataset OfficeHome --source-domain RealWorld --target-domain Product \
    --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

